import sys,os
import numpy as np
sys.path.append(os.pardir) #为了能导入父目录中的文件而进行的设定
from dataset.mnist import load_mnist
from PIL import Image #Python Imaging Library（PIL）是Python的一个图像处理库


def img_show(img):
    # 把保存为NumPy数组的图像数据转换为PIL用
    pil_img = Image.fromarray(np.uint8(img))
    pil_img.show()

if __name__ == '__main__':
    # normalize 标准化&归一化
    # flatten 拉平
    (x_train, t_train), (x_test, t_test) = load_mnist(normalize=False,
                                                      flatten=True,
                                                    )
    print(x_train.shape) #（60000, 784）
    print(t_train.shape) #（60000,）
    print(x_test.shape) #（10000, 784）
    print(t_test.shape) # （10000,）
    img  = x_train[1]
    label = t_train[1]
    print(label)
    print(img.shape)
    img = img.reshape(28,28)
    print(img.shape)

    img_show(img)